package com.controller;


import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.alibaba.fastjson.JSONObject;

import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Collection;

import javax.servlet.ServletException;
import javax.servlet.annotation.MultipartConfig;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.Part;

//使用@WebServlet配置UploadServlet的访问路径
@WebServlet(name="UploadServlet",urlPatterns="/UploadServlet")
//使用注解@MultipartConfig将一个Servlet标识为支持文件上传
@MultipartConfig//标识Servlet支持文件上传
public class Upload extends javax.servlet.http.HttpServlet {

    protected void doPost(javax.servlet.http.HttpServletRequest request, javax.servlet.http.HttpServletResponse response) throws javax.servlet.ServletException, IOException {
        request.setCharacterEncoding("utf-8");
        response.setCharacterEncoding("utf-8");
        response.setContentType("text/html;charset=utf-8");
        JSONObject json = new JSONObject();
        //存储路径/毕业设计管理子系统layui-javaweb版/WebContent/upload
        String src="/zhaopin/upload/";
        String src2="../../upload/";
//        String src2="../upload/";
//        String savePath=" F:\\PROJECT\\JAVA PROJECT\\zhaopin\\web\\upload";
//        F:\PROJECT\JAVA PROJECT\zhaopin
        String savePath = request.getServletContext().getRealPath("/upload/");
        System.out.println("savePath = " + savePath);
        //获取上传的文件集合
        Collection<Part> parts = request.getParts();
        System.out.println("parts = " + parts.size());
        //上传单个文件
        if (parts.size()==1) {
            //Servlet3.0将multipart/form-data的POST请求封装成Part，通过Part对上传的文件进行操作。
            //Part part = parts[0];//从上传的文件集合中获取Part对象
            Part part = request.getPart("file");//通过表单file控件(<input type="file" name="file">)的名字直接获取Part对象
            //Servlet3没有提供直接获取文件名的方法,需要从请求头中解析出来
            //获取请求头，请求头的格式：form-data; name="file"; filename="snmp4j--api.zip"
            System.out.println("part = " + part);
            String header = part.getHeader("content-disposition");
            System.out.println("header = " + header);
            //获取文件名
            String fileName = getFileName(header);
            System.out.println("fileName = " + fileName);
            //把文件写到指定路径
//            part.write(savePath+File.separator+fileName);
            part.write(savePath+fileName);
            System.out.println("savePath+fileName = " + savePath+fileName);
            json.put("msg","上传成功");
            json.put("code",0);
            json.put("src",src+fileName);
            json.put("src2",src2+fileName);
            System.out.println("json = " + json);
        }else {
            //一次性上传多个文件
            for (Part part : parts) {//循环处理上传的文件
                //获取请求头，请求头的格式：form-data; name="file"; filename="snmp4j--api.zip"
                String header = part.getHeader("content-disposition");
                //获取文件名
                String fileName = getFileName(header);
                //把文件写到指定路径
                part.write(savePath+File.separator+fileName);
                json.put("msg","上传失败");
                json.put("code",0);
                json.put("src",src);
                json.put("src2",src2);
            }
        }
        PrintWriter out = response.getWriter();
        /* out.println("success"); */
        out.println(json);
        System.out.println("json = " + json);
        out.flush();
        out.close();
    }

    /**
     * 根据请求头解析出文件名
     * 请求头的格式：火狐和google浏览器下：form-data; name="file"; filename="snmp4j--api.zip"
     *                 IE浏览器下：form-data; name="file"; filename="E:\snmp4j--api.zip"
     * @param header 请求头
     * @return 文件名
     */
    public String getFileName(String header) {
        /**
         * String[] tempArr1 = header.split(";");代码执行完之后，在不同的浏览器下，tempArr1数组里面的内容稍有区别
         * 火狐或者google浏览器下：tempArr1={form-data,name="file",filename="snmp4j--api.zip"}
         * IE浏览器下：tempArr1={form-data,name="file",filename="E:\snmp4j--api.zip"}
         */
        String[] tempArr1 = header.split(";");
        /**
         *火狐或者google浏览器下：tempArr2={filename,"snmp4j--api.zip"}
         *IE浏览器下：tempArr2={filename,"E:\snmp4j--api.zip"}
         */
        String[] tempArr2 = tempArr1[2].split("=");
        //获取文件名，兼容各种浏览器的写法
        String fileName = tempArr2[1].substring(tempArr2[1].lastIndexOf("\\")+1).replaceAll("\"", "");
        return fileName;
    }

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        doPost(req, resp);
    }
}